Traceback (most recent call last):
  File "/home/sjain441/results/jax-robotarium-simulation/qmix/seed 2/qmix-at-best/main.py", line 60, in <module>
    qvals, hs = actor(obs, hs)
                ^^^^^^^^^^^^^^
  File "/home/sjain441/miniconda3/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/sjain441/miniconda3/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/sjain441/results/jax-robotarium-simulation/qmix/seed 2/qmix-at-best/actor.py", line 47, in forward
    hidden = self.GRUCell_0(embedding, hidden)
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/sjain441/miniconda3/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/sjain441/miniconda3/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/sjain441/results/jax-robotarium-simulation/qmix/seed 2/qmix-at-best/actor.py", line 24, in forward
    r_t = torch.sigmoid(x @ self.W_r + h_prev @ self.U_r + self.b_r)
                                                           ^^^^^^^^
  File "/home/sjain441/miniconda3/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1915, in __getattr__
    def __getattr__(self, name: str) -> Union[Tensor, "Module"]:

KeyboardInterrupt
